#!/bin/bash
# 摘要：启用 Python 自动命令补齐功能
# 描述：设置 Python 使用 readline 和 rlcompleter 模块实现自动命令补齐。
# 需要导入 readline 和 rlcompleter 模块。

TAB_SCRIPT="/usr/bin/tab.py"
PYTHONSTARTUP_LINE="export PYTHONSTARTUP=${TAB_SCRIPT}"

# 检查是否存在 tab.py 脚本文件；如果不存在，则创建
if [ ! -f "${TAB_SCRIPT}" ]; then
    cat << EOF > "${TAB_SCRIPT}"
import readline
import rlcompleter
readline.parse_and_bind('tab: complete')
EOF
fi

# 如果 /etc/profile 中未添加 PYTHONSTARTUP 变量，则添加
if ! grep -qF "${PYTHONSTARTUP_LINE}" /etc/profile; then
    echo "${PYTHONSTARTUP_LINE}" | sudo tee -a /etc/profile
fi

# 重新加载 /etc/profile 以应用更改
source /etc/profile